package com.lagou.edu.factory;

import com.alibaba.druid.util.StringUtils;
import com.lagou.edu.annotation.*;
import com.lagou.edu.dao.AccountDao;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import java.io.File;
import java.io.InputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;

/**
 * @author 应癫
 * <p>
 * 工厂类，生产对象（使用反射技术）
 */
public class BeanFactory {

    /**
     * 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
     * 任务二：对外提供获取实例对象的接口（根据id获取）
     */

    private static Map<String, Object> map = new HashMap<>();  // 存储对象

    private static List<String> classPaths = new ArrayList<String>();

    static {
        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
        // 加载xml
        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
        // 解析xml
        SAXReader saxReader = new SAXReader();
        try {
            Document document = saxReader.read(resourceAsStream);
            Element rootElement = document.getRootElement();
            // 获取scan标签
            List<Element> scanList = rootElement.selectNodes("//scan");

            for (int i = 0; i < scanList.size(); i++) {
                Element element = scanList.get(i);
                // 获取要扫描的包路径
                String basePackage = element.attributeValue("base-package");        // com.lagou.edu
                //先把包名转换为路径,首先得到项目的classpath
                String classpath = BeanFactory.class.getResource("/").getPath();
                //然后把我们的包名basPach转换为路径名
                basePackage = basePackage.replace(".", File.separator);
                //然后把classpath和basePack合并
                String searchPath = classpath + basePackage;
                doPath(new File(searchPath));
                //这个时候我们已经得到了指定包下所有的类的绝对路径了。我们现在利用这些绝对路径和java的反射机制得到他们的类对象
                for (String s : classPaths) {
                    //把 D:\work\code\20170401\search-class\target\classes\com\baibin\search\a\A.class 这样的绝对路径转换为全类名com.baibin.search.a.A
                    s = s.replace(classpath.replace("/", "\\").replaceFirst("\\\\", ""), "").replace("\\", ".").replace(".class", "");
                    Class cls = Class.forName(s);
                    // 判断注解
                    Repository repositoryannotation = (Repository) cls.getDeclaredAnnotation(Repository.class);
                    String value = "";
                    if (repositoryannotation != null) {
                        value = repositoryannotation.value();
                    }
                    Service serviceannotation = (Service) cls.getDeclaredAnnotation(Service.class);
                    if (serviceannotation != null) {
                        value = serviceannotation.value();
                    }

                    Component componentannotation = (Component) cls.getDeclaredAnnotation(Component.class);
                    if (repositoryannotation != null || serviceannotation != null || componentannotation != null) {
                        Object o = cls.newInstance();  // 实例化之后的对象
                        // 没有指定了名称
                        if (StringUtils.isEmpty(value)) {
                            // 判断是否实现了接口
                            if (cls.getInterfaces().length > 0) {
                                value = cls.getInterfaces()[0].getSimpleName();
                            } else {
                                value = cls.getSimpleName();
                            }
                            value = new StringBuilder().append(Character.toLowerCase(value.charAt(0))).append(value.substring(1)).toString();
                        }
                        // 存储到map中待用
                        map.put(value, o);
                    }
                }
            }

            // 依赖的属性赋值
            Set<Map.Entry<String, Object>> entries = map.entrySet();
            for (Map.Entry<String, Object> entry : entries) {
                Object object = entry.getValue();
                Class<?> aClass = object.getClass();
                Field[] declaredFields = aClass.getDeclaredFields();
                // 循环判读获取有注解的属性
                for (Field field : declaredFields) {
                    Autowired[] annotationsByType = field.getAnnotationsByType(Autowired.class);
                    if (annotationsByType.length > 0) {
                        Class type = field.getType();
                        // 循环判断根据类型注入
                        for (Map.Entry<String, Object> entry1 : entries) {
                            Object o = entry1.getValue();
                            Class aClass1 = entry1.getValue().getClass();
                            // 判断类型关系
                            if (type.isAssignableFrom(aClass1)){
                                field.setAccessible(true);
                                field.set(object, o);
                            }
                        }
                    }
                }
            }

            // 判断方法是否用了事务注解，如果用了事务注解，则生成代理类
            Set<Map.Entry<String, Object>> entries1 = map.entrySet();
            for (Map.Entry<String, Object> entry : entries1) {
                Object value = entry.getValue();
                Class<?> aClass = value.getClass();
                Method[] methods = aClass.getMethods();
                for (Method method: methods) {
                    Transactional annotation = method.getAnnotation(Transactional.class);
                    if (annotation != null){
                        ProxyFactory proxyFactory = (ProxyFactory)map.get("proxyFactory");
                        // 判断是否实现了接口
                        if (aClass.getInterfaces().length > 0) {
                            Object cglibProxy = proxyFactory.getJdkProxy(value);
                            map.put(entry.getKey(),cglibProxy);
                            break;
                        } else {
                            Object cglibProxy = proxyFactory.getCglibProxy(value);
                            map.put(entry.getKey(),cglibProxy);
                            break;
                        }

                    }
                }
            }




            /*List<Element> beanList = rootElement.selectNodes("//bean");
            for (int i = 0; i < beanList.size(); i++) {
                Element element = beanList.get(i);
                // 处理每个bean元素，获取到该元素的id 和 class 属性
                String id = element.attributeValue("id");        // accountDao
                String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
                // 获取class对象k
                Class<?> aClass = Class.forName(clazz);
                //

                Object o = aClass.newInstance();  // 实例化之后的对象

                // 存储到map中待用
                map.put(id, o);

            }

            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
            // 有property子元素的bean就有传值需求
            List<Element> propertyList = rootElement.selectNodes("//property");
            // 解析property，获取父元素
            for (int i = 0; i < propertyList.size(); i++) {
                Element element = propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
                String name = element.attributeValue("name");
                String ref = element.attributeValue("ref");

                // 找到当前需要被处理依赖关系的bean
                Element parent = element.getParent();

                // 调用父元素对象的反射功能
                String parentId = parent.attributeValue("id");
                Object parentObject = map.get(parentId);
                // 遍历父对象中的所有方法，找到"set" + name
                Method[] methods = parentObject.getClass().getMethods();
                for (int j = 0; j < methods.length; j++) {
                    Method method = methods[j];
                    if (method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
                        method.invoke(parentObject, map.get(ref));
                    }
                }

                // 把处理之后的parentObject重新放到map中
                map.put(parentId, parentObject);

            }*/


        } catch (DocumentException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InstantiationException e) {
            e.printStackTrace();
        }/* catch (InvocationTargetException e) {
            e.printStackTrace();
        }*/

    }


    // 任务二：对外提供获取实例对象的接口（根据id获取）
    public static Object getBean(String id) {
        return map.get(id);
    }

    /**
     * 该方法会得到所有的类，将类的绝对路径写入到classPaths中
     *
     * @param file
     */
    private static void doPath(File file) {
        if (file.isDirectory()) {//文件夹
            //文件夹我们就递归
            File[] files = file.listFiles();
            for (File f1 : files) {
                doPath(f1);
            }
        } else {//标准文件
            //标准文件我们就判断是否是class文件
            if (file.getName().endsWith(".class")) {
                //如果是class文件我们就放入我们的集合中。
                classPaths.add(file.getPath());
            }
        }
    }
}
